import os
import shutil

if __name__ == '__main__':
    folder = os.path.exists("temp_img")
    if not folder:  # 判断是否存在文件夹如果不存在则创建为文件夹
        os.makedirs("temp_img")  # makedirs 创建文件时如果路径不存在会创建这个路径
    else:
        shutil.rmtree("temp_img")
        os.makedirs("temp_img")

    print("请输入有裂痕图片文件夹路径（文件夹内不允许存在其他文件）")
    root_path = input()
    crack_names = os.listdir(root_path)

    for i, img_name in enumerate(crack_names):
        img_path = os.path.join(root_path, img_name)
        shutil.copyfile(img_path, f"temp_img/crack_{i}.jpg")

    print("请输入无裂痕图片文件夹路径（文件夹内不允许存在其他文件）, 连续输入路径，输入N/n结束")
    normal_names = []
    while True:
        root_path = input()
        if root_path.upper() == "N":
            break
        else:
            normal_names += [os.path.join(root_path, name) for name in os.listdir(root_path)]

    for i, img_name in enumerate(normal_names):
        shutil.copyfile(img_name, f"temp_img/normal_{i}.jpg")

